from simpletransformers.classification.classification_model import ClassificationModel

TransformerModel = ClassificationModel
